# class Count:
#     def add(self, a, b):
#         return a+b

class Count:
    def add(self):
        pass


from unittest import mock
import unittest


class test_add(unittest.TestCase):
    def test_add(self):
        count = Count()
        # 找到要替换的对象，这个对象可以是一个类，或者是一个函数，或者是一个类实例
        # 实例化Mock类得到一个mock对象，并且设置这个mock对象的行为，比如被调用的时候返回什么值，被访问成员的时候返回什么值等
        count.add = mock.Mock(return_value=12)
        result = count.add(8, 4)
        self.assertEqual(result, 12)


if __name__ == '__main__':
    unittest.main()
